home *** CD-ROM | disk | FTP | other *** search
/ Languguage OS 2 / Languguage OS II Version 10-94 (Knowledge Media)(1994).ISO / gnu / glibc108.zip / glibc108 / sunrpc / svc_tcp.c < prev    next >
C/C++ Source or Header  |  1994-02-06  |  10KB  |  420 lines
  1. /* @(#)svc_tcp.c    2.2 88/08/01 4.0 RPCSRC */
  2. /*
  3.  * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
  4.  * unrestricted use provided that this legend is included on all tape
  5.  * media and as a part of the software program in whole or part.  Users
  6.  * may copy or modify Sun RPC without charge, but are not authorized
  7.  * to license or distribute it to anyone else except as part of a product or
  8.  * program developed by the user.
  9.  * 
  10.  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
  11.  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
  12.  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
  13.  * 
  14.  * Sun RPC is provided with no support and without any obligation on the
  15.  * part of Sun Microsystems, Inc. to assist in its use, correction,
  16.  * modification or enhancement.
  17.  * 
  18.  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
  19.  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
  20.  * OR ANY PART THEREOF.
  21.  * 
  22.  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
  23.  * or profits or other special, indirect and consequential damages, even if
  24.  * Sun has been advised of the possibility of such damages.
  25.  * 
  26.  * Sun Microsystems, Inc.
  27.  * 2550 Garcia Avenue
  28.  * Mountain View, California  94043
  29.  */
  30. #if !defined(lint) && defined(SCCSIDS)
  31. static char sccsid[] = "@(#)svc_tcp.c 1.21 87/08/11 Copyr 1984 Sun Micro";
  32. #endif
  33.  
  34. /*
  35.  * svc_tcp.c, Server side for TCP/IP based RPC. 
  36.  *
  37.  * Copyright (C) 1984, Sun Microsystems, Inc.
  38.  *
  39.  * Actually implements two flavors of transporter -
  40.  * a tcp rendezvouser (a listner and connection establisher)
  41.  * and a record/tcp stream.
  42.  */
  43.  
  44. #include <stdio.h>
  45. #include <rpc/rpc.h>
  46. #include <sys/socket.h>
  47. #include <errno.h>
  48. extern bool_t abort();
  49. extern errno;
  50.  
  51. /*
  52.  * Ops vector for TCP/IP based rpc service handle
  53.  */
  54. static bool_t        svctcp_recv();
  55. static enum xprt_stat    svctcp_stat();
  56. static bool_t        svctcp_getargs();
  57. static bool_t        svctcp_reply();
  58. static bool_t        svctcp_freeargs();
  59. static void        svctcp_destroy();
  60.  
  61. static struct xp_ops svctcp_op = {
  62.     svctcp_recv,
  63.     svctcp_stat,
  64.     svctcp_getargs,
  65.     svctcp_reply,
  66.     svctcp_freeargs,
  67.     svctcp_destroy
  68. };
  69.  
  70. /*
  71.  * Ops vector for TCP/IP rendezvous handler
  72.  */
  73. static bool_t        rendezvous_request();
  74. static enum xprt_stat    rendezvous_stat();
  75.  
  76. static struct xp_ops svctcp_rendezvous_op = {
  77.     rendezvous_request,
  78.     rendezvous_stat,
  79.     abort,
  80.     abort,
  81.     abort,
  82.     svctcp_destroy
  83. };
  84.  
  85. static int readtcp(), writetcp();
  86. static SVCXPRT *makefd_xprt();
  87.  
  88. struct tcp_rendezvous { /* kept in xprt->xp_p1 */
  89.     u_int sendsize;
  90.     u_int recvsize;
  91. };
  92.  
  93. struct tcp_conn {  /* kept in xprt->xp_p1 */
  94.     enum xprt_stat strm_stat;
  95.     u_long x_id;
  96.     XDR xdrs;
  97.     char verf_body[MAX_AUTH_BYTES];
  98. };
  99.  
  100. /*
  101.  * Usage:
  102.  *    xprt = svctcp_create(sock, send_buf_size, recv_buf_size);
  103.  *
  104.  * Creates, registers, and returns a (rpc) tcp based transporter.
  105.  * Once *xprt is initialized, it is registered as a transporter
  106.  * see (svc.h, xprt_register).  This routine returns
  107.  * a NULL if a problem occurred.
  108.  *
  109.  * If sock<0 then a socket is created, else sock is used.
  110.  * If the socket, sock is not bound to a port then svctcp_create
  111.  * binds it to an arbitrary port.  The routine then starts a tcp
  112.  * listener on the socket's associated port.  In any (successful) case,
  113.  * xprt->xp_sock is the registered socket number and xprt->xp_port is the
  114.  * associated port number.
  115.  *
  116.  * Since tcp streams do buffered io similar to stdio, the caller can specify
  117.  * how big the send and receive buffers are via the second and third parms;
  118.  * 0 => use the system default.
  119.  */
  120. SVCXPRT *
  121. svctcp_create(sock, sendsize, recvsize)
  122.     register int sock;
  123.     u_int sendsize;
  124.     u_int recvsize;
  125. {
  126.     bool_t madesock = FALSE;
  127.     register SVCXPRT *xprt;
  128.     register struct tcp_rendezvous *r;
  129.     struct sockaddr_in addr;
  130.     int len = sizeof(struct sockaddr_in);
  131.  
  132.     if (sock == RPC_ANYSOCK) {
  133.         if ((sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
  134.             perror("svctcp_.c - udp socket creation problem");
  135.             return ((SVCXPRT *)NULL);
  136.         }
  137.         madesock = TRUE;
  138.     }
  139.     bzero((char *)&addr, sizeof (addr));
  140.     addr.sin_family = AF_INET;
  141.     if (bindresvport(sock, &addr)) {
  142.         addr.sin_port = 0;
  143.         (void)bind(sock, (struct sockaddr *)&addr, len);
  144.     }
  145.     if ((getsockname(sock, (struct sockaddr *)&addr, &len) != 0)  ||
  146.         (listen(sock, 2) != 0)) {
  147.         perror("svctcp_.c - cannot getsockname or listen");
  148.         if (madesock)
  149.                (void)close(sock);
  150.         return ((SVCXPRT *)NULL);
  151.     }
  152.     r = (struct tcp_rendezvous *)mem_alloc(sizeof(*r));
  153.     if (r == NULL) {
  154.         (void) fprintf(stderr, "svctcp_create: out of memory\n");
  155.         return (NULL);
  156.     }
  157.     r->sendsize = sendsize;
  158.     r->recvsize = recvsize;
  159.     xprt = (SVCXPRT *)mem_alloc(sizeof(SVCXPRT));
  160.     if (xprt == NULL) {
  161.         (void) fprintf(stderr, "svctcp_create: out of memory\n");
  162.         return (NULL);
  163.     }
  164.     xprt->xp_p2 = NULL;
  165.     xprt->xp_p1 = (caddr_t)r;
  166.     xprt->xp_verf = _null_auth;
  167.     xprt->xp_ops = &svctcp_rendezvous_op;
  168.     xprt->xp_port = ntohs(addr.sin_port);
  169.     xprt->xp_sock = sock;
  170.     xprt_register(xprt);
  171.     return (xprt);
  172. }
  173.  
  174. /*
  175.  * Like svtcp_create(), except the routine takes any *open* UNIX file
  176.  * descriptor as its first input.
  177.  */
  178. SVCXPRT *
  179. svcfd_create(fd, sendsize, recvsize)
  180.     int fd;
  181.     u_int sendsize;
  182.     u_int recvsize;
  183. {
  184.  
  185.     return (makefd_xprt(fd, sendsize, recvsize));
  186. }
  187.  
  188. static SVCXPRT *
  189. makefd_xprt(fd, sendsize, recvsize)
  190.     int fd;
  191.     u_int sendsize;
  192.     u_int recvsize;
  193. {
  194.     register SVCXPRT *xprt;
  195.     register struct tcp_conn *cd;
  196.  
  197.     xprt = (SVCXPRT *)mem_alloc(sizeof(SVCXPRT));
  198.     if (xprt == (SVCXPRT *)NULL) {
  199.         (void) fprintf(stderr, "svc_tcp: makefd_xprt: out of memory\n");
  200.         goto done;
  201.     }
  202.     cd = (struct tcp_conn *)mem_alloc(sizeof(struct tcp_conn));
  203.     if (cd == (struct tcp_conn *)NULL) {
  204.         (void) fprintf(stderr, "svc_tcp: makefd_xprt: out of memory\n");
  205.         mem_free((char *) xprt, sizeof(SVCXPRT));
  206.         xprt = (SVCXPRT *)NULL;
  207.         goto done;
  208.     }
  209.     cd->strm_stat = XPRT_IDLE;
  210.     xdrrec_create(&(cd->xdrs), sendsize, recvsize,
  211.         (caddr_t)xprt, readtcp, writetcp);
  212.     xprt->xp_p2 = NULL;
  213.     xprt->xp_p1 = (caddr_t)cd;
  214.     xprt->xp_verf.oa_base = cd->verf_body;
  215.     xprt->xp_addrlen = 0;
  216.     xprt->xp_ops = &svctcp_op;  /* truely deals with calls */
  217.     xprt->xp_port = 0;  /* this is a connection, not a rendezvouser */
  218.     xprt->xp_sock = fd;
  219.     xprt_register(xprt);
  220.     done:
  221.     return (xprt);
  222. }
  223.  
  224. static bool_t
  225. rendezvous_request(xprt)
  226.     register SVCXPRT *xprt;
  227. {
  228.     int sock;
  229.     struct tcp_rendezvous *r;
  230.     struct sockaddr_in addr;
  231.     int len;
  232.  
  233.     r = (struct tcp_rendezvous *)xprt->xp_p1;
  234.     again:
  235.     len = sizeof(struct sockaddr_in);
  236.     if ((sock = accept(xprt->xp_sock, (struct sockaddr *)&addr,
  237.         &len)) < 0) {
  238.         if (errno == EINTR)
  239.             goto again;
  240.            return (FALSE);
  241.     }
  242.     /*
  243.      * make a new transporter (re-uses xprt)
  244.      */
  245.     xprt = makefd_xprt(sock, r->sendsize, r->recvsize);
  246.     xprt->xp_raddr = addr;
  247.     xprt->xp_addrlen = len;
  248.     return (FALSE); /* there is never an rpc msg to be processed */
  249. }
  250.  
  251. static enum xprt_stat
  252. rendezvous_stat()
  253. {
  254.  
  255.     return (XPRT_IDLE);
  256. }
  257.  
  258. static void
  259. svctcp_destroy(xprt)
  260.     register SVCXPRT *xprt;
  261. {
  262.     register struct tcp_conn *cd = (struct tcp_conn *)xprt->xp_p1;
  263.  
  264.     xprt_unregister(xprt);
  265.     (void)close(xprt->xp_sock);
  266.     if (xprt->xp_port != 0) {
  267.         /* a rendezvouser socket */
  268.         xprt->xp_port = 0;
  269.     } else {
  270.         /* an actual connection socket */
  271.         XDR_DESTROY(&(cd->xdrs));
  272.     }
  273.     mem_free((caddr_t)cd, sizeof(struct tcp_conn));
  274.     mem_free((caddr_t)xprt, sizeof(SVCXPRT));
  275. }
  276.  
  277. /*
  278.  * All read operations timeout after 35 seconds.
  279.  * A timeout is fatal for the connection.
  280.  */
  281. static struct timeval wait_per_try = { 35, 0 };
  282.  
  283. /*
  284.  * reads data from the tcp conection.
  285.  * any error is fatal and the connection is closed.
  286.  * (And a read of zero bytes is a half closed stream => error.)
  287.  */
  288. static int
  289. readtcp(xprt, buf, len)
  290.     register SVCXPRT *xprt;
  291.     caddr_t buf;
  292.     register int len;
  293. {
  294.     register int sock = xprt->xp_sock;
  295. #ifdef FD_SETSIZE
  296.     fd_set mask;
  297.     fd_set readfds;
  298.  
  299.     FD_ZERO(&mask);
  300.     FD_SET(sock, &mask);
  301. #else
  302.     register int mask = 1 << sock;
  303.     int readfds;
  304. #endif /* def FD_SETSIZE */
  305.     do {
  306.         readfds = mask;
  307.         if (select(_rpc_dtablesize(), &readfds, (int*)NULL, (int*)NULL, 
  308.                &wait_per_try) <= 0) {
  309.             if (errno == EINTR) {
  310.                 continue;
  311.             }
  312.             goto fatal_err;
  313.         }
  314. #ifdef FD_SETSIZE
  315.     } while (!FD_ISSET(sock, &readfds));
  316. #else
  317.     } while (readfds != mask);
  318. #endif /* def FD_SETSIZE */
  319.     if ((len = read(sock, buf, len)) > 0) {
  320.         return (len);
  321.     }
  322. fatal_err:
  323.     ((struct tcp_conn *)(xprt->xp_p1))->strm_stat = XPRT_DIED;
  324.     return (-1);
  325. }
  326.  
  327. /*
  328.  * writes data to the tcp connection.
  329.  * Any error is fatal and the connection is closed.
  330.  */
  331. static int
  332. writetcp(xprt, buf, len)
  333.     register SVCXPRT *xprt;
  334.     caddr_t buf;
  335.     int len;
  336. {
  337.     register int i, cnt;
  338.  
  339.     for (cnt = len; cnt > 0; cnt -= i, buf += i) {
  340.         if ((i = write(xprt->xp_sock, buf, cnt)) < 0) {
  341.             ((struct tcp_conn *)(xprt->xp_p1))->strm_stat =
  342.                 XPRT_DIED;
  343.             return (-1);
  344.         }
  345.     }
  346.     return (len);
  347. }
  348.  
  349. static enum xprt_stat
  350. svctcp_stat(xprt)
  351.     SVCXPRT *xprt;
  352. {
  353.     register struct tcp_conn *cd =
  354.         (struct tcp_conn *)(xprt->xp_p1);
  355.  
  356.     if (cd->strm_stat == XPRT_DIED)
  357.         return (XPRT_DIED);
  358.     if (! xdrrec_eof(&(cd->xdrs)))
  359.         return (XPRT_MOREREQS);
  360.     return (XPRT_IDLE);
  361. }
  362.  
  363. static bool_t
  364. svctcp_recv(xprt, msg)
  365.     SVCXPRT *xprt;
  366.     register struct rpc_msg *msg;
  367. {
  368.     register struct tcp_conn *cd =
  369.         (struct tcp_conn *)(xprt->xp_p1);
  370.     register XDR *xdrs = &(cd->xdrs);
  371.  
  372.     xdrs->x_op = XDR_DECODE;
  373.     (void)xdrrec_skiprecord(xdrs);
  374.     if (xdr_callmsg(xdrs, msg)) {
  375.         cd->x_id = msg->rm_xid;
  376.         return (TRUE);
  377.     }
  378.     return (FALSE);
  379. }
  380.  
  381. static bool_t
  382. svctcp_getargs(xprt, xdr_args, args_ptr)
  383.     SVCXPRT *xprt;
  384.     xdrproc_t xdr_args;
  385.     caddr_t args_ptr;
  386. {
  387.  
  388.     return ((*xdr_args)(&(((struct tcp_conn *)(xprt->xp_p1))->xdrs), args_ptr));
  389. }
  390.  
  391. static bool_t
  392. svctcp_freeargs(xprt, xdr_args, args_ptr)
  393.     SVCXPRT *xprt;
  394.     xdrproc_t xdr_args;
  395.     caddr_t args_ptr;
  396. {
  397.     register XDR *xdrs =
  398.         &(((struct tcp_conn *)(xprt->xp_p1))->xdrs);
  399.  
  400.     xdrs->x_op = XDR_FREE;
  401.     return ((*xdr_args)(xdrs, args_ptr));
  402. }
  403.  
  404. static bool_t
  405. svctcp_reply(xprt, msg)
  406.     SVCXPRT *xprt;
  407.     register struct rpc_msg *msg;
  408. {
  409.     register struct tcp_conn *cd =
  410.         (struct tcp_conn *)(xprt->xp_p1);
  411.     register XDR *xdrs = &(cd->xdrs);
  412.     register bool_t stat;
  413.  
  414.     xdrs->x_op = XDR_ENCODE;
  415.     msg->rm_xid = cd->x_id;
  416.     stat = xdr_replymsg(xdrs, msg);
  417.     (void)xdrrec_endofrecord(xdrs, TRUE);
  418.     return (stat);
  419. }
  420.